import numpy as np
import pandas as pd
import networkx as nx
import matplotlib.pyplot as plt
import xlrd
import readExcel

#nodes = np.arange(49)
nodes=[
    '0.0'
    '1.0',
    '2.0',
    '3.0',
    '4.0',
    '5.0',
    '6.0',
    '7.0',
    '8.0',
    '9.0',
    '10.0','11.0','12.0','13.0','14.0','15.0','16.0','17.0','18.0','19.0'
    '20.0','21.0','22.0','23.0','24.0','25.0','26.0','27.0','28.0','29.0'
    '30.0','31.0','32.0','33.0','34.0','35.0','36.0','37.0','38.0','39.0'
    '40.0','41.0','42.0','43.0','44.0','45.0','46.0','47.0','48.0',
]
edges = readExcel.excel()


G = nx.Graph()

for node in nodes:
    G.add_node(node)

G.add_edges_from(edges)


#nx.draw(G, with_labels=True, node_color='y')
nx.draw(G,pos = nx.random_layout(G),node_color = 'b',edge_color = 'r',with_labels = True, font_size =18,node_size =20)
#plt.show()
plt.savefig('2.png', dpi=400, bbox_inches='tight')

#https://blog.csdn.net/weixin_42815846/article/details/113313044
#https://blog.csdn.net/qq951127336/article/details/54586869/